package com.ruoyi.system.controller;

import com.baidubce.qianfan.Qianfan;
import com.baidubce.qianfan.model.chat.ChatResponse;
import com.ruoyi.common.core.controller.BaseController;
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping("/qianfan")
public class QianfanController extends BaseController
{
    private final Qianfan _qianfan = new Qianfan(
            "79bc4870039f43b4a2c2a3a2bc05bf39",
            "5c8d0680226f4ee89d28b6869f4dfd92"
    );

    @GetMapping("/random/plan")
    public String random_plan()
    {
        ChatResponse resp = _qianfan.chatCompletion()
                .model("ERNIE-3.5-8K")
                .temperature(1.0)
                .topP(1.0)
                .addMessage("user",
                        "你是一名流体仿真平台开发工程师，" +
                                "你现在需要生成一项流体仿真计划数据用于测试，" +
                                "报告计划要有创造性，前沿而新奇，" +
                                "需要包括计划名称(name)，计划描述(description), 计划参数(args)，" +
                                "计划参数以文字描述形式给出，整体控制在200字符以内。" +
                                "只以json格式给出返回结果，不要输入任何额外内容")
                .execute();
        String res = resp.getResult();
        return res.substring(7, res.length()-3);
    }

}
